package com.indusfo.spc.controller;

import com.indusfo.spc.aspect.SystemControllerLog;
import com.indusfo.spc.pojo.ExcelErrorData;
import com.indusfo.spc.pojo.Pro;
import com.indusfo.spc.pojo.Station;
import com.indusfo.spc.service.ProService;
import com.indusfo.spc.untlis.ExcelUtil;
import com.indusfo.spc.untlis.GsonUtil;
import com.indusfo.spc.vo.JSONObject;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.List;

/**
 * 工序前端控制层
 *
 * @author xuz
 * @date 2018/11/23 1:44 PM
 */
@Controller
@RequestMapping("/service/pro")
public class ProController {

    @Autowired
    private ProService proService;


    /**
     * 更新
     *
     * @return com.indusfo.spc.vo.JSONObject
     * @author xuz
     * @date 2018/11/29 11:19 AM
     */
    @RequestMapping(value = "/updatePro", method = RequestMethod.POST)
    @SystemControllerLog(description = "更新工序信息")
    @ResponseBody
    public JSONObject updatePro(Pro pro, Integer[] staIds) {

        JSONObject json = proService.updatePro(pro, staIds);
        return json;
    }

    /**
     * 插入
     *
     * @param pro
     * @return com.indusfo.spc.vo.JSONObject
     * @author xuz
     * @date 2018/11/29 11:19 AM
     */
    @RequestMapping(value = "/insertPro", method = RequestMethod.POST)
    @SystemControllerLog(description = "新增工序信息")
    @ResponseBody
    public JSONObject insertPro(Pro pro, Integer[] staIds) {

        JSONObject json = proService.insertPro(pro, staIds);
        return json;
    }

    /**
     * 查询
     *
     * @return com.indusfo.spc.vo.JSONObject
     * @author xuz
     * @date 2018/11/29 11:19 AM
     */
    @RequestMapping(value = "/queryPro")
    @SystemControllerLog(description = "查询工序信息")
    @ResponseBody
    public JSONObject queryPro(Pro pro) {

        JSONObject json = proService.queryPro(pro);
        return json;
    }

    /**
     * 查询  排除数组中的数据
     */
    @RequestMapping(value = "/queryFilterProList")
    @ResponseBody
    public JSONObject queryFilterProList(Pro pro, Long[] proIds) {
        JSONObject json = proService.queryProIds(pro, proIds);
        return json;
    }

    /**
     * 查询  排除数组中的数据
     */
    @RequestMapping(value = "/queryBadPhenoProList")
    @ResponseBody
    public JSONObject queryBadPhenoProList(Pro pro, Integer badphenoId) {
        JSONObject json = proService.queryBadPhenoProList(pro, badphenoId);
        return json;
    }

    @RequestMapping(value = "/definesPro")
    @SystemControllerLog(description = "启用工序信息")
    @ResponseBody
    public JSONObject definesPro(Long[] proIds, Integer dataState) {
        JSONObject json = proService.deletePro(proIds, dataState);
        return json;
    }


    @RequestMapping("/uploadFilePro")
    @ResponseBody
    public JSONObject uploadFileUser(@RequestParam("file") MultipartFile file, Boolean way, HttpServletRequest request, HttpServletResponse response) throws IOException, ParseException {

        //定义一个message对象用于向前台传输数据
        JSONObject jsonObject = new JSONObject();
        jsonObject.setStatus(JSONObject.SUCCESS);

        //设置错误集合的
        List<ExcelErrorData> errorDatas = new ArrayList<>();
        response.setContentType("text/html;charset=UTF-8");
        //判断上传文件格式是否正确
        Workbook wb = ExcelUtil.getWorkbook(file.getInputStream(), ExcelUtil.isExcel2007(file.getOriginalFilename()));
        if (wb == null) {
            errorDatas.add(ExcelErrorData.getExcelErrorData(0, 0, "文件解析错误"));
        } else {
            //获取excel文件的第一个sheet
            Sheet sheet = wb.getSheetAt(0);
            if (sheet == null) {
                jsonObject.setStatus(JSONObject.ERROR);
                errorDatas.add(ExcelErrorData.getExcelErrorData(0, 0, "未找到需要导入的表格"));
            } else {
                //获取最后一行的行号
                int rowCount = sheet.getLastRowNum();
                if (rowCount < 1) {
                    jsonObject.setStatus(JSONObject.ERROR);
                    errorDatas.add(ExcelErrorData.getExcelErrorData(0, "未找到需要导入的数据，请确认表格是否保存！！！"));
                } else {
                    List<Pro> proList = new ArrayList<>();
                    for (int i = 1; i <= rowCount; i++) {
                        boolean needAdd = true;
                        //获取到每一行的每一列的数据
                        //工序名称
                        String vcProName = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 0, false);
                        //工序类型
                        String vcItemName = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 1, false);
                        //备注
                        String vcRemark = com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, i, 2, false);
                        Pro pro = new Pro(vcItemName, vcRemark, vcProName);
                        int m = i + 1;
                        //将excel中的每一行数据存储进行检查
                        String result = proService.checkNoRepeat(pro);
                        if (result != null) {
                            errorDatas.add(ExcelErrorData.getExcelErrorData(i, result));
                            needAdd = false;
                        }
                        for (int j = 1; j < i; j++) {
                            //判断工序名称是否重复
                            String oldvcProName = StringUtils.trimToEmpty(com.indusfo.spc.untlis.ExcelUtil.getCellValue(sheet, j, 0, false)).replaceAll("\\s*", "");
                            if (StringUtils.isBlank(vcProName) && oldvcProName.equals(vcProName)) {
                                needAdd = false;
                                errorDatas.add(ExcelErrorData.getExcelErrorData(i, "与表格中第" + j + "行存在相同工序名称"));
                            }
                        }
                        if (needAdd) {
                            proList.add(pro);
                        } else {
                            jsonObject.setStatus(JSONObject.ERROR);
                            jsonObject.setMsg("导入数据中存在异常信息");
                        }


                    }
                    //根据这些来判断是否是中止导入还是跳过继续进行导入
                    //根据这些来判断是否是中止导入还是跳过继续进行导入
                    if (jsonObject.getStatus().equals(JSONObject.ERROR) && !way) {
                        jsonObject.setMsg(jsonObject.getMsg() + ",中止导入");
                    } else {
                        if (proList.size() > 0) {
                            try {
                                proService.updateListPro(proList);
                                //当程序执行到这里时,判断信息是否为空为空的话设置导入成功
                                jsonObject.setMsg(StringUtils.isBlank(jsonObject.getMsg()) ? "导入成功" : jsonObject.getMsg());
                                if ("导入成功".equals(jsonObject.getMsg())) {
                                    jsonObject.setStatus(JSONObject.SUCCESS);
                                }
                            } catch (Exception e) {
                                e.printStackTrace();
                                jsonObject.setMsg("导入失败");
                            }
                        }
                    }
                    //msg.setData(GsonUtil.toJson(errorDatas));
                }
            }
        }
        jsonObject.setData(GsonUtil.toJson(errorDatas));
        return JSONObject.build(JSONObject.SUCCESS, jsonObject.getMsg(), errorDatas);
    }


    /**
     * 去添加查询（产品，任务单，批次）
     */
    @RequestMapping(value = "/queryListProToInsert")
    @ResponseBody
    public JSONObject queryListProToInsert(Pro pro) {
        JSONObject json = proService.queryListProToInsert(pro);
        return json;
    }
}
